package main

import (
	"fmt"
	"os"
)

func startMenu() {
	fmt.Println("欢迎来到学院信息管理系统")
	fmt.Println("1、添加学员")
	fmt.Println("2、编辑学员信息")
	fmt.Println("3、展示所有学员信息")
	fmt.Println("4、退出系统")

}

func getInput() *student {
	var (
		id    int
		name  string
		class string
	)
	fmt.Print("请按要求输入学员信息\n")
	fmt.Print("请输入学员id：")
	fmt.Scan(&id)
	fmt.Print("请输入学员姓名：")
	fmt.Scan(&name)
	fmt.Print("请输入学员班级：")
	fmt.Scan(&class)
	stu := newStudent(id, name, class)
	return stu

}
func main() {
	sm := newStuMgr()
	for {
		startMenu()

		var input int
		fmt.Print("请输入你要操作的序号：")
		fmt.Scan(&input)
		switch input {
		case 1:
			stu := getInput()
			sm.addStu(stu)
		case 2:
			stu := getInput()
			sm.editStu(stu)
		case 3:
			sm.viewStu()
		case 4:
			os.Exit(0)
		}
	}
}
